package com.bigmodel.baidu;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * markdown格式文本解析工具
 */
public class MarkdownUtil {

    public static final String JAVA_PATTERN = "\\```java(.*?)\\```";
    public static final String JSON_PATTERN = "\\```json(.*?)\\```";

    /**
     * 从markdown文本中提取json文本
     *
     * @param text markdown文本
     * @return 提取到的json文本
     */
    public static String extractJavaText(String text) {
        // 使用正则表达式提取JSON数据部分
        Pattern pattern = Pattern.compile(JAVA_PATTERN, Pattern.DOTALL);
        Matcher matcher = pattern.matcher(text);
        String json = null;
        if (matcher.find()) {
            json = matcher.group(1).trim();
        }
        return json;
    }

    public static String extractJsonText(String text) {
        // 使用正则表达式提取JSON数据部分
        Pattern pattern = Pattern.compile(JSON_PATTERN, Pattern.DOTALL);
        Matcher matcher = pattern.matcher(text);
        String json = null;
        if (matcher.find()) {
            json = matcher.group(1).trim();
        }
        return json;
    }
}
